check if element present in the list or not using list comprehension in python